package com.cb2.algorithm.leetcode;

/**
 * <a href='https://leetcode.cn/problems/maximum-number-of-vowels-in-a-substring-of-given-length'>定长子串中元音的最大数目(Maximum Number of Vowels in a Substring of Given Length)</a>
 * <p>给你字符串 s 和整数 k 。</p>
 * <p>请返回字符串 s 中长度为 k 的单个子字符串中可能包含的最大元音字母数。</p>
 * <p>英文中的 元音字母 为（a, e, i, o, u）。</p>
 *
 * <p>
 * <b>示例：</b>
 * <pre>
 *  示例 1：
 *      输入：s = "abciiidef", k = 3
 *      输出：3
 *      解释：子字符串 "iii" 包含 3 个元音字母。
 *
 *  示例 2：
 *      输入：s = "aeiou", k = 2
 *      输出：2
 *      解释：任意长度为 2 的子字符串都包含 2 个元音字母。
 *
 *  示例 3：
 *      输入：s = "leetcode", k = 3
 *      输出：2
 *      解释："lee"、"eet" 和 "ode" 都包含 2 个元音字母。
 *
 *  示例 4：
 *      输入：s = "rhythms", k = 4
 *      输出：0
 *      解释：字符串 s 中不含任何元音字母。
 *
 *  示例 5：
 *      输入：s = "tryhard", k = 4
 *      输出：1
 * </pre>
 * </p>
 *
 * <p>
 * <b>提示：</b>
 * <ul>
 *     <li>1 <= s.length <= 10^5</li>
 *     <li>s 由小写英文字母组成</li>
 *     <li>1 <= k <= s.length</li>
 * </ul>
 * </p>
 *
 * @author c2b
 * @since 2025/4/28 16:59
 */
public class LC1456MaximumNumberOfVowelsInSubstringOfGivenLength_M {
    static class Solution {
        public int maxVowels(String s, int k) {
            byte[] table = new byte[128];
            table['a'] = 1;
            table['e'] = 1;
            table['i'] = 1;
            table['o'] = 1;
            table['u'] = 1;

            int res = 0;
            int vowel_cnt = 0;
            char[] chars = s.toCharArray();
            // 小优化：提前退出
            for (int i = 0; i < s.length() && res < k; i++) {
                vowel_cnt += table[chars[i]];
                if (i >= k - 1) {
                    // 能执行到这里，窗口形成，统计结果
                    res = Math.max(res, vowel_cnt);
                    // 移除窗口最左侧的元素
                    vowel_cnt -= table[(chars[i - k + 1])];
                }
            }
            return res;
        }

        public int maxVowels2(String s, int k) {
            char[] chars = s.toCharArray();
            // 形成窗口
            int vowel_cnt = 0;
            for (int i = 0; i < k; i++) {
                vowel_cnt += isVowel(chars[i]);
            }
            int res = vowel_cnt;
            for (int i = k; i < s.length() && res < k; i++) {
                // 窗口向右移动
                vowel_cnt -= isVowel(chars[i - k]);
                vowel_cnt += isVowel(chars[i]);
                // 更新结果
                res = Math.max(res, vowel_cnt);
            }
            return res;
        }

        private int isVowel(char c) {
            return c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u' ? 1 : 0;
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        System.out.println(solution.maxVowels("abciiidef", 3));
        System.out.println(solution.maxVowels("aeiou", 2));
        System.out.println(solution.maxVowels("leetcode", 3));
        System.out.println(solution.maxVowels("rhythms", 4));
        System.out.println(solution.maxVowels("tryhard", 4));
        System.out.println(solution.maxVowels("weallloveyou", 7));
    }
}
